package random;

import com.sun.xml.internal.fastinfoset.tools.XML_SAX_StAX_FI;

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;

/**
 * 分发糖果
 * 老师想给孩子们分发糖果，有 N个孩子站成了一条直线，老师会根据每个孩子的表现，预先给他们评分。
 * 你需要按照以下要求，帮助老师给这些孩子分发糖果：
 *          每个孩子至少分配到 1 个糖果。
 *          评分更高的孩子必须比他两侧的邻位孩子获得更多的糖果。
 * 那么这样下来，老师至少需要准备多少颗糖果呢？
 */

public class L135 {

    public static void main(String[] args) {

        System.out.println(candy(new int[]{1,3,2,2,1}));
    }

    public static int candy(int[] ratings) {

        // 先创建一个map来记录每个位置是否派了糖果
        Map<Integer,Integer> map = new HashMap<>();
        for (int i = 0; i < ratings.length; i++) {
            map.put(i, 1);
        }

        for (int i = 1; i < ratings.length; i++) {
            if (ratings[i] > ratings[i-1]) {
                map.put(i, map.get(i-1)+1);
            }
        }

        for (int i = ratings.length - 1; i > 0; i--) {
            if (ratings[i] < ratings[i-1]) {
                map.put(i-1, Math.max(map.get(i) + 1, map.get(i-1)));
            }
        }

        // 遍历map
        int count = 0;
        Set<Integer> integers = map.keySet();
        for (Integer i : integers) {
            System.out.println(i + ":" + map.get(i));
            count += map.get(i);
        }
        return count;
    }
}
